package com.zgx.plus.system.service.impl;

import cn.hutool.core.convert.Convert;
import cn.hutool.core.lang.Assert;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zgx.plus.common.aop.DataSource;
import com.zgx.plus.common.constant.ComConstant;
import com.zgx.plus.common.constant.DataSourceType;
import com.zgx.plus.common.exception.BizException;
import com.zgx.plus.common.redis.RedisCache;
import com.zgx.plus.common.redis.RedisConstant;
import com.zgx.plus.system.api.pos.SysConfig;
import com.zgx.plus.system.api.vos.SysConfigVo;
import com.zgx.plus.system.mapper.SysConfigMapper;
import com.zgx.plus.system.service.SysConfigService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import java.text.MessageFormat;
import java.util.Collection;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.function.Supplier;

@Service
public class SysConfigServiceImpl extends ServiceImpl<SysConfigMapper, SysConfig> implements SysConfigService {

    @Autowired
    private RedisCache redisCache;

    @Override
    public String fetchConfigByKey(String key) {
        return Optional.of(this.getOne(new LambdaQueryWrapper<SysConfig>().eq(SysConfig::getConfigKey, key)))
                .map(SysConfig::getConfigValue).orElse(null);
    }

    /**
     * 项目启动时，初始化参数到缓存
     */
    @PostConstruct
    public void init() {
        loadingConfigCache();
    }

    /**
     * 查询参数配置信息
     *
     * @param configId 参数配置ID
     * @return 参数配置信息
     */
    @Override
    @DataSource(DataSourceType.MASTER)
    public SysConfig selectConfigById(Integer configId) {
        return this.getById(configId);
    }

    /**
     * 根据键名查询参数配置信息
     *
     * @param configKey 参数key
     * @return 参数键值
     */
    @Override
    public String selectConfigByKey(String configKey) {
        String configValue = Convert.toStr(redisCache.getCacheObject(getCacheKey(configKey)));
        if (StrUtil.isNotEmpty(configValue)) {
            return configValue;
        }
        configValue = this.getObj(Wrappers.lambdaQuery(SysConfig.class).select(SysConfig::getConfigValue)
                .eq(SysConfig::getConfigKey, configKey).last("limit 1"), val -> {
            if (Objects.isNull(val)) {
                return StrUtil.EMPTY;
            } else {
                String value = String.valueOf(val);
                redisCache.setCacheObject(getCacheKey(configKey), value);
                return value;
            }
        });
        return configValue;
    }

    /**
     * 获取验证码开关
     *
     * @return true开启，false关闭
     */
    @Override
    public boolean selectCaptchaEnabled() {
        String captchaEnabled = selectConfigByKey("sys.account.captchaEnabled");
        if (StrUtil.isEmpty(captchaEnabled)) {
            return true;
        }
        return Convert.toBool(captchaEnabled, false);
    }

    /**
     * 查询参数配置列表
     *
     * @param config 参数配置信息
     * @return 参数配置集合
     */
    @Override
    public List<SysConfig> selectConfigList(SysConfigVo config) {
        String beginTime = String.valueOf(config.getParams().getOrDefault(ComConstant.BEGIN_TIME, StrUtil.EMPTY));
        String endTime = String.valueOf(config.getParams().getOrDefault(ComConstant.END_TIME, StrUtil.EMPTY));
        return this.list(Wrappers.lambdaQuery(SysConfig.class)
                .eq(StrUtil.isNotBlank(config.getConfigType()), SysConfig::getConfigType, config.getConfigType())
                .like(StrUtil.isNotBlank(config.getConfigKey()), SysConfig::getConfigKey, config.getConfigKey())
                .like(StrUtil.isNotBlank(config.getConfigName()), SysConfig::getConfigName, config.getConfigName())
                .ge(StrUtil.isNotBlank(beginTime), SysConfig::getCreateTime, beginTime)
                .le(StrUtil.isNotBlank(endTime), SysConfig::getCreateTime, endTime));
    }

    /**
     * 新增参数配置
     *
     * @param config 参数配置信息
     * @return 结果
     */
    @Override
    public boolean insertConfig(SysConfig config) {
        boolean bSave = this.save(config);
        if (bSave) {
            redisCache.setCacheObject(getCacheKey(config.getConfigKey()), config.getConfigValue());
        }
        return bSave;
    }

    /**
     * 修改参数配置
     *
     * @param config 参数配置信息
     * @return 结果
     */
    @Override
    public boolean updateConfig(SysConfig config) {
        SysConfig temp = this.getById(config.getConfigId());
        if (!StrUtil.equals(temp.getConfigKey(), config.getConfigKey())) {
            redisCache.deleteObject(getCacheKey(temp.getConfigKey()));
        }
        boolean updateCode = this.updateById(config);
        if (updateCode) {
            redisCache.setCacheObject(getCacheKey(config.getConfigKey()), config.getConfigValue());
        }
        return updateCode;
    }

    /**
     * 批量删除参数信息
     *
     * @param configIds 需要删除的参数ID
     */
    @Override
    public void deleteConfigByIds(Integer[] configIds) {
        for (Integer configId : configIds) {
            SysConfig config = selectConfigById(configId);
            Assert.notEquals(ComConstant.YES, config.getConfigType(),
                    () -> new BizException(MessageFormat.format("内置参数{0}不能删除 ", config.getConfigKey())));
            this.removeById(configId);
            redisCache.deleteObject(getCacheKey(config.getConfigKey()));
        }
    }

    /**
     * 加载参数缓存数据
     */
    @Override
    public void loadingConfigCache() {
        this.list(Wrappers.lambdaQuery(SysConfig.class).select(SysConfig::getConfigKey, SysConfig::getConfigType))
                .stream().filter(config -> Objects.nonNull(config.getConfigKey()))
                .forEach(config -> redisCache.setCacheObject(getCacheKey(config.getConfigKey()), config.getConfigValue()));
    }

    /**
     * 清空参数缓存数据
     */
    @Override
    public void clearConfigCache() {
        Collection<String> keys = redisCache.keys(RedisConstant.SYS_CONFIG_KEY + "*");
        redisCache.deleteObject(keys);
    }

    /**
     * 重置参数缓存数据
     */
    @Override
    public void resetConfigCache() {
        clearConfigCache();
        loadingConfigCache();
    }

    /**
     * 校验参数键名是否唯一
     *
     * @param config 参数配置信息
     * @return 结果
     */
    @Override
    public boolean checkConfigKeyUnique(SysConfig config) {

        Integer id = this.getObj(Wrappers.lambdaQuery(SysConfig.class).select(SysConfig::getConfigId)
                        .eq(SysConfig::getConfigKey, config.getConfigKey()).last("limit 1"),
                configId -> Objects.isNull(configId) ? -1 : Integer.parseInt(String.valueOf(configId)));
        return Objects.equals(id, config.getConfigId()) ? ComConstant.NOT_UNIQUE : ComConstant.UNIQUE;
    }

    /**
     * 设置cache key
     *
     * @param configKey 参数键
     * @return 缓存键key
     */
    private String getCacheKey(String configKey) {
        return RedisConstant.SYS_CONFIG_KEY + configKey;
    }
}
